cd ../..

gpus=$1

python -m trainer.CLTrainer fit \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/desed_with_id.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --model.exp_name="3_baselines/pure_CL/proj_1e-3" \
    --data.init_args.batch_sizes=[0,0,12,0,192] \
    --trainer.devices=${gpus}  \
    --trainer.strategy="ddp_find_unused_parameters_true" \
    --trainer.max_epochs=10000